package net.mingsoft.document.util;

import net.mingsoft.document.constant.FileConstant;

/**
 * @author gengfa
 * @date 22/02/15
 */
public class FileTypeUtil {

    private FileTypeUtil() {}

    public static String getType(String ext) {
        if (FileTypeUtil.isImage(ext)) {
            return FileConstant.IMAGE;
        } else if (FileTypeUtil.isWord(ext)) {
            return FileConstant.WORD;
        } else if (FileTypeUtil.isExcel(ext)) {
            return FileConstant.EXCEL;
        } else if (FileTypeUtil.isPpt(ext)) {
            return FileConstant.PPT;
        } else if (FileTypeUtil.isPdf(ext)) {
            return FileConstant.PDF;
        } else {
            return FileConstant.OTHER;
        }
    }

    public static boolean isImage(String ext) {
        for (String s : FileConstant.IMAGE_EXT) {
            if (s.equals(ext)) {
                return true;
            }
        }
        return false;
    }

    public static boolean isWord(String ext) {
        for (String s : FileConstant.WORD_EXT) {
            if (s.equals(ext)) {
                return true;
            }
        }
        return false;
    }

    public static boolean isExcel(String ext) {
        for (String s : FileConstant.EXCEL_EXT) {
            if (s.equals(ext)) {
                return true;
            }
        }
        return false;
    }

    public static boolean isPpt(String ext) {
        for (String s : FileConstant.PPT_EXT) {
            if (s.equals(ext)) {
                return true;
            }
        }
        return false;
    }

    public static boolean isPdf(String ext) {
        for (String s : FileConstant.PDF_EXT) {
            if (s.equals(ext)) {
                return true;
            }
        }
        return false;
    }
}
